"""
GDP动态柱状图开发
"""
from pyecharts.charts import Bar,Timeline
from pyecharts.options import *
from pyecharts.globals import *
# 读取数据
file_name = "D:/python code/1、Python快速入门（8天零基础入门到精通）/第1-12章资料/可视化案例数据/动态柱状图数据/1960-2019全球GDP数据.csv"
f = open(file_name,"r",encoding="GB2312")
data_lines = f.readlines()
# 关闭文件
f.close()
# 删除第一行用不到的数据
data_lines.pop(0)

# 将数据转为字典存储，格式为：
# { 年份：[[国家，gdp],[国家，GDP],......], 年份：[[国家，gdp],[国家，GDP],......], ......}
data_dict = {}
for data_line in data_lines:
    year = int(data_line.split(",")[0])
    country = data_line.split(",")[1]
    gdp = float(data_line.split(",")[2])
    try:
        data_dict[year].append([country,gdp])
    except:
        data_dict[year]=[]
        data_dict[year].append([country,gdp])

# 排序年份
gdp_sorted_data = sorted(data_dict.keys())

# 构建时间线对象
timeline = Timeline(
    {"theme":ThemeType.CHALK}
)
# 处理每一年里面的数据
for year in gdp_sorted_data:
    # 按照gdp值进行排序
    data_dict[year].sort(key=lambda element:element[1],reverse=True)
    # 并取出前8名
    year_data=data_dict[year][0:8]
    x_data = []
    y_data = []
    for country_gdp in year_data:
        x_data.append(country_gdp[0])  # x轴
        y_data.append(country_gdp[1]/100000000)  # y轴
    # X轴和Y轴数据反转
    x_data.reverse()
    y_data.reverse()
    #创建每一个柱状图对象
    bar = Bar()
    bar.add_xaxis(xaxis_data=x_data)
    bar.add_yaxis("GDP(亿)",y_data,label_opts=LabelOpts(position="right"))
    bar.reversal_axis()
    bar.set_global_opts(
        title_opts=TitleOpts(title=f"{year}年全球前8GDP数据")
    )

    timeline.add(bar,str(year))
    timeline.add_schema(
        play_interval=1000,
        is_timeline_show=True,
        is_auto_play=True,
        is_loop_play=False
    )



# 绘图
timeline.render("1960-2019全球GDP前8国家.html")
